package yunjiao.javatutorials.guava.concurrent;

import com.google.common.util.concurrent.SimpleTimeLimiter;
import com.google.common.util.concurrent.TimeLimiter;
import com.google.common.util.concurrent.UncheckedTimeoutException;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

/**
 * 代理模式（全局控制类的所有方法）
 *
 * 适合需要对某个类的所有方法统一设置超时的场景（如第三方 SDK 客户端）。
 *
 * @author yangyunjiao
 */
public class TimeLimiterDemo {
    public static void main(String[] args) {
        // 3. 创建 TimeLimiter 实例（默认使用 daemon 线程池，也可自定义线程池）
        ExecutorService executor = Executors.newSingleThreadExecutor();
        TimeLimiter timeLimiter = SimpleTimeLimiter.create(executor);

        // 4. 生成代理对象（对 TaskService 的所有方法设置超时）
        TaskService proxy = timeLimiter.newProxy(
                new TaskService.TaskServiceImpl(), // 目标对象
                TaskService.class,    // 目标接口
                2,                    // 超时时间
                TimeUnit.SECONDS      // 时间单位
        );

        // 5. 调用代理方法（超时会抛出 TimeoutException）
        try {
            String result = proxy.longRunningTask("测试参数");
            System.out.println(result); // 若 2 秒内完成则输出结果
        } catch (UncheckedTimeoutException e) {
            System.err.println("任务超时（2秒）：" + e.getMessage());
        } finally {
            executor.shutdown(); // 关闭线程池（避免资源泄露）
        }
    }
}
